from torch import nn
import torch,math,random
import collections
import torch.nn.functional as F
import numpy as np

print(torch.__version__)
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")

print(device)
